package xin.xueqing.base.controller;


import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.map.MapUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.api.ApiController;
import com.baomidou.mybatisplus.extension.api.R;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import xin.xueqing.base.dto.BaseDeptDto;
import xin.xueqing.base.po.BaseDeptPO;
import xin.xueqing.base.service.BaseDeptService;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.io.Serializable;
import java.util.List;
import java.util.stream.Collectors;

/**
 * 部门表(BaseDept)表控制层
 *
 * @author xueqing wang
 * @since 2021-03-08 22:49:11
 */
@RestController
@RequestMapping("/adminApi/depts")
public class BaseDeptController extends ApiController {
    /**
     * 服务对象
     */
    @Resource
    private BaseDeptService baseDeptService;

    /**
     * 分页查询所有数据
     *
     * @param page     分页对象
     * @param baseDept 查询实体
     * @return 所有数据
     */
    @GetMapping
    public R selectPage(Page<BaseDeptPO> page, BaseDeptPO baseDept) {
        return success(this.baseDeptService.page(page, new QueryWrapper<>(baseDept)));
    }

    /**
     * 查询所有数据
     *
     * @param baseDept 查询实体
     * @return 所有数据
     */
    @GetMapping("/selectAll")
    public R selectAll(BaseDeptPO baseDept) {
        return success(this.baseDeptService.list(new QueryWrapper<>(baseDept)));
    }

    /**
     * 查询所有数据
     *
     * @param baseDept 查询实体
     * @return 所有数据
     */
    @GetMapping("/selectChildren")
    public R selectChildren(BaseDeptPO baseDept) {
        baseDept.setParentId(0);
        List<BaseDeptPO> list = this.baseDeptService.list(new QueryWrapper<>(baseDept));
        List<BaseDeptDto> collect = list.stream().map(baseDeptPO ->
                BaseDeptDto.builder().key(baseDeptPO.getId()).title(baseDeptPO.getName()).build()
        ).collect(Collectors.toList());

        for (BaseDeptDto baseDeptDto : collect) {
            baseDeptDto =  transfer(baseDeptDto);
        }
        return success(collect);
    }

    BaseDeptDto transfer(BaseDeptDto baseDeptDto) {
        List<BaseDeptPO> list = baseDeptService.list(new QueryWrapper<BaseDeptPO>().eq("parent_id", baseDeptDto.getKey()));

        if (CollUtil.isNotEmpty(list)) {
            List<BaseDeptDto> collect = list.stream().map(baseDeptPO ->
                    BaseDeptDto.builder().key(baseDeptPO.getId()).title(baseDeptPO.getName()).build()
            ).collect(Collectors.toList());

            baseDeptDto.setChildren(collect);
            for (BaseDeptDto deptDto : collect) {
                deptDto = transfer(deptDto);
            }

            return baseDeptDto;
        } else {
            return baseDeptDto;
        }
    }


    /**
     * 通过主键查询单条数据
     *
     * @param id 主键
     * @return 单条数据
     */
    @GetMapping("{id}")
    public R selectOne(@PathVariable Serializable id) {
        return success(this.baseDeptService.getById(id));
    }

    /**
     * 新增数据
     *
     * @param baseDept 实体对象
     * @return 新增结果
     */
    @PostMapping
    public R insert(@RequestBody BaseDeptPO baseDept) {
        return success(this.baseDeptService.save(baseDept));
    }

    /**
     * 修改数据
     *
     * @param baseDept 实体对象
     * @return 修改结果
     */
    @PutMapping
    public R update(@RequestBody BaseDeptPO baseDept) {
        return success(this.baseDeptService.updateById(baseDept));
    }

    /**
     * 删除数据
     *
     * @param idList 主键结合
     * @return 删除结果
     */
    @DeleteMapping
    public R delete(@RequestParam("idList") List<Long> idList) {
        return success(this.baseDeptService.removeByIds(idList));
    }
}
